convert image to matrix python